package com.jia.controller;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.commons.CommonsMultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;

@RestController
public class FileController {

    @RequestMapping("/upload")
    public String fileUpload(@RequestParam("file")CommonsMultipartFile file, HttpServletRequest request) throws IOException {
        // 获取文件名
        String uploadFileName = file.getOriginalFilename();
        // 如果文件名为空，直接返回首页
        if("".equals(uploadFileName)){
            return "redirect:/index.jsp";
        }
        System.out.println("上传文件名为："+uploadFileName);
        // 上传路径保存
        String path = request.getServletContext().getRealPath("/upload");
        // 如果文件路径不存在，创建一个
        File realPath = new File(path);
        if(!realPath.exists()){
            realPath.mkdir();
        }
        System.out.println("文件保存的地址为："+realPath);

        InputStream is = file.getInputStream(); //文件输入流
        FileOutputStream os = new FileOutputStream(new File(realPath, uploadFileName)); //文件输出流
        // 读取写出
        int len = 0;
        byte[] buffer = new byte[1024];
        while ((len=is.read(buffer))!=-1){
            os.write(buffer,0,len);
            os.flush();
        }
        os.close();
        is.close();

        return "redirect:/index.jsp";
    }

    @RequestMapping("/upolad2")
    public String fileUpload2(@RequestParam("file") CommonsMultipartFile file,HttpServletRequest request) throws IOException{
        // 设置上传路径
        String path = request.getServletContext().getRealPath("/upload");
        File realPath = new File(path);
        if(!realPath.exists()){
            realPath.mkdir();
        }
        // 上传文件地址
        System.out.println("上传文件保存地址："+realPath);
        // 通过CommonsMultipartFile直接写文件
        file.transferTo(new File(realPath+"/"+file.getOriginalFilename()));

        return "redirect:/index.jsp";
    }

    @RequestMapping("/download")
    public String downloads(HttpServletResponse response,HttpServletRequest request) throws IOException{
        // 要下载的图片地址
        String path = request.getServletContext().getRealPath("/upload");
        String fileName = "xxx.jpg";

        //设置response 响应头
        response.reset(); //设置页面不缓存
        response.setCharacterEncoding("UTF-8");
        response.setContentType("multipart/form-data"); //二进制数据传输
        response.setHeader("Content-Disposition",
                "attachment;fileName="+ URLEncoder.encode(fileName,"UTF-8"));
        File file = new File(path, fileName);
        FileInputStream input = new FileInputStream(file);
        ServletOutputStream out = response.getOutputStream();
        int index = 0;
        byte[] buffer = new byte[1024];
        // 执行写操作
        while ((index = input.read(buffer))!=-1){
            out.write(buffer,0,index);
            out.flush();
        }
        out.close();
        input.close();

        return "ok";
    }
}
